#pragma once

#include <pthread.h>
#include <string>
#include <functional>

namespace ThreadModule
{
    using func_t = std::function<void(std::string&)>; // 外部想要让线程执行的函数

    class Thread
    {
    private:
        void Excute()
        {
            _is_running = true;
            _func(_name);
            _is_running = false;
        }

        static void *ThreadRoutine(void *args)
        {
            Thread *self = static_cast<Thread *>(args);
            self->Excute();
            return nullptr;
        }

    public:
        Thread(const std::string name, func_t func)
            : _name(name), _func(func), _is_running(false)
        {
        }

        // 启动线程
        bool start()
        {
            int n = pthread_create(&_tid, nullptr, ThreadRoutine, this);
            return n == 0 ? true : false;
        }

        // 终止线程
        void stop()
        {
            if (_is_running)
            {
                pthread_cancel(_tid);
                _is_running = false;
            }
        }

        // 等待线程
        void join()
        {
            ::pthread_join(_tid, nullptr);
        }

        // 获取线程的名称
        std::string get_name()
        {
            return _name;
        }

        // 获取线程的状态
        std::string get_status()
        {
            return _is_running == true ? "start" : "stop";
        }

        ~Thread()
        {
        }

    private:
        std::string _name; // 线程名字
        pthread_t _tid;    // 线程 tid
        func_t _func;   // 线程执行的任务
        bool _is_running;  // 线程的运行状态
    };
}